// Copyright 2022 jeff.li. and/or its affiliates.

#pragma once

#include <tbir/pipeline/global_unique_index.h>
#include <tbir/pipeline/op_kernel.h>

namespace tbir::runtime {

    class PyTorchInferOp : public OpKernel {
    public:
        void Init() override;

        RTValue Process(PyArgs inputs) const override;

    protected:
        OpKernelPtr impl_;

        friend class TXSession;
    };

}  // namespace tbir::runtime
